from api import log
from config import HOST


class ApiRegisterLogin:
    # 初始化
    def __init__(self, session):
        self.session = session

        self.url_img_code = HOST + "/common/public/verifycode1/{}"
        self.url_phone_code = HOST + "/member/public/sendSms"
        self.url_register = HOST + "/member/public/reg"
        self.url_login = HOST + "/member/public/login"
        self.url_login_status = HOST + "/member/public/islogin"

    # 1、获取图片验证码接口 封装
    def api_img_code(self, random):
        log.info("正在调用获取图片验证码接口，请求方法：get 请求url：{}".format(self.url_img_code.format(random)))
        return self.session.get(url=self.url_img_code.format(random))

    # 2、获取短信验证码接口 封装
    def api_phone_code(self, phone, imgVerifyCode):
        data = {
            "phone": phone,
            "imgVerifyCode": imgVerifyCode,
            "type": "reg"
        }
        log.info("正在调用获取短信验证码接口，请求方法:post 请求url:{} 请求数据:{}".format(self.url_phone_code,data))
        return self.session.post(url=self.url_phone_code, data=data)

    # 3、注册接口 封装
    def api_register(self, phone, password,  verifycode, phone_code):
        data = {
            "phone": phone,
            "password": password,
            "verifycode": verifycode,
            "phone_code": phone_code,
            "dy_server": "on",
            "invite_phone": ""
        }
        log.info("正在调用注册接口，请求方法：post 请求url：{} 请求数据:{}".format(self.url_register, data))
        return self.session.post(url=self.url_register, data=data)

    # 4、登录接口 封装
    def api_login(self, keywords="15037447712", password="y123456"):
        data = {
            "keywords": keywords,
            "password": password,
        }
        log.info("正在调用登录接口，请求方法：post 请求url：{} 请求数据:{}".format(self.url_login, data))
        return self.session.post(url=self.url_login, data=data)

    # 5、查询登陆状态接口 封装
    def api_login_status(self):
        log.info("正在调用登录状态接口，请求方法：post 请求url：{}".format(self.url_login_status))
        return self.session.post(url=self.url_login_status)